#!/bin/bash
sbatch <<EOT
#!/bin/sh
#SBATCH --job-name=$3
#SBATCH -o ./log/$3.%A.out
#SBATCH -e ./log/$3.%A.err
#SBATCH -p debug
#SBATCH --gres=gpu:$2
#SBATCH -c1
#SBATCH --mem=5G
#SBATCH --time=2-00:00:00
#SBATCH --nodelist=$1

# Set-up the environment.
source ${HOME}/.bashrc
conda activate torch110

set PYTHONPATH=./

# Start the experiment.
python ./$3.py --num_gpus=$2
EOT
